package com.study.www.service;

import com.study.www.config.MyJob;
import com.study.www.enums.ConfigEnum;
import com.study.www.model.Config;
import org.quartz.*;
import org.quartz.spi.MutableTrigger;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.quartz.SchedulerFactoryBean;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @author : 89007337.zhuyanpeng
 * @date :Create in  2019/8/3- 22:15
 **/
@Component
public class QuartzTableManager {

    @Autowired
    SchedulerFactoryBean schedulerFactoryBean;

    /***************************  此三者均可复用，类似于 Factory 一样。故提出做公用 *********************/
    //调度器
    private static Scheduler scheduler;
    //JobBuilder
    private static JobBuilder jobBuilder;
    //触发器Builder
    private static TriggerBuilder<Trigger> triggerBuilder ;


    //毫秒的间隔时间
    private static final Integer MILLISECONDS=1000;

    @PostConstruct
    public void init(){
        //因为Job 被自己给实现了故此处应去指定对应的Job类去创建一个调度器
        QuartzTableManager.jobBuilder= JobBuilder.newJob(MyJob.class);
        QuartzTableManager.scheduler = schedulerFactoryBean.getScheduler();
        triggerBuilder = TriggerBuilder.newTrigger();
    }

    /**
     * 增加任务
     *
     * @param :com.study.www.model.config
     * @Date: 2018/2/26 9:57
     * @return: void
     */
     void addJob(Config config) throws SchedulerException {
        //得到调度器
        JobKey jobKey = this.getJobKey(config);
        //获得触发器
        TriggerKey triggerKey = TriggerKey.triggerKey(config.getName(), config.getGroup());
        Trigger trigger = scheduler.getTrigger(triggerKey);
        //判断触发器是否存在（如果存在说明之前运行过但是在当前被禁用了，如果不存在说明一次都没运行过）
        if (trigger == null) {
            //新建一个工作任务 指定任务类型为串接进行的
            JobDetail jobDetail = jobBuilder.withIdentity(jobKey).build();
            //将工作添加到工作任务当中去
            JobDataMap jobDataMap = jobDetail.getJobDataMap();
            jobDataMap.put(ConfigEnum.SCHEDULEJOB.getCode(), config);
            trigger = getTrigger(config, triggerKey);
            //在调度器中将触发器和任务进行组合
            scheduler.scheduleJob(jobDetail, trigger);
        } else {
            //按照新的规则进行
            trigger = getTrigger(config, triggerKey);
            //重启
            scheduler.rescheduleJob(triggerKey, trigger);
        }
    }
    
    /**
    * 注意 simpleTrigger 和 cronTrigger 互斥只可以选择一个
    */
    private  Trigger getTrigger(Config config,TriggerKey triggerKey ){
        Trigger trigger =null;
        if(config.getCount() == null){
            //将cron表达式进行转换
            CronScheduleBuilder cronScheduleBuilder = CronScheduleBuilder.cronSchedule(config.getCron());
            trigger = triggerBuilder.withIdentity(triggerKey).withSchedule(cronScheduleBuilder).build();
        }else{
            //指定的时间和时间段来进行定时任务
            MutableTrigger build = SimpleScheduleBuilder.simpleSchedule()
                    //执行次数
                    .withRepeatCount(config.getCount())
                    //间隔时间 秒
                    .withIntervalInMilliseconds(config.getIntervalSecond() * MILLISECONDS)
                    .build();
            //开始执行时间
            build.setStartTime(config.getStartTime());
            //结束执行时间
            build.setEndTime(config.getEndTime());
            //绑定 组信息
            build.setKey(triggerKey);
            trigger = (Trigger)build;
        }
        return trigger;
    }

    /**
     * 删除任务
     *
     * @param : com.study.www.model.config
     * @Date: 2018/2/24 18:23
     * @return: void
     */
     void deleteJob(Config config) throws SchedulerException {
        //找到key值
        JobKey jobKey = this.getJobKey(config);
        //从触发器找到此任务然后进行删除
        scheduler.deleteJob(jobKey);
    }

    /**
     * 根据name和group得到任务的key
     *
     * @param :com.study.www.model.config
     * @Date: 2018/2/24 18:27
     * @return: org.quartz.JobKey
     */
     JobKey getJobKey(Config config) {
        return getJobKey(config.getName(), config.getGroup());
    }

     JobKey getJobKey(String name, String group) {
        return JobKey.jobKey(name, group);
    }
}
